// Copyright (c) Microsoft Corporation.
// Licensed under the MIT License.

import { IExplanationDashboardData } from "@responsible-ai/core-ui";

export const ebmData: IExplanationDashboardData = {
  dataSummary: {
    classNames: [">50k", "not >50k"],
    featureNames: ["Age", "Employment"]
  },
  modelInformation: { method: "classifier", modelClass: "blackbox" },
  precomputedExplanations: {
    ebmGlobalExplanation: {
      feature_list: [
        {
          lower_bounds: [
            -1.9630728974955807, -1.9630728974955807, -1.9630728974955807,
            -1.9630728974955807, -1.9154099165913767, -1.7806141340403567,
            -1.7764739856623946, -1.4100443273717669, -0.776687301262559,
            -0.7352827682778065, -0.4896034940116783, -0.39413713261449346,
            -0.15192057469367137, 0.01601427118353499, 0.011283153457547566,
            0.020374194354029505, 0.06597892018491641, 0.2581146647220247,
            0.20190220524579094, 0.32974146711947017, 0.4446291402225885,
            0.4802105623130199, 0.5140535634900176, 0.5534187551955145,
            0.5055305546863824, 0.5052638226242903, 0.37892315475582405,
            0.5824786970134876, 0.6662451536716028, 0.730826955789492,
            0.8339124550988825, 0.8340819457577475, 0.7904756213671229,
            0.7953369454803966, 0.7271482107266113, 0.9252469007948072,
            0.9342798905918867, 0.7870009670424168, 0.642822876834768,
            0.6529310684715242, 0.6427683889770234, 0.6173548277470535,
            0.810725119316934, 0.636707172061967, 0.5390435422682236,
            0.36642499466232054, 0.2559695472968398, 0.24718392103637093,
            0.01862004584014733, -0.033898271846272324, -0.015609731291912982,
            0.13576034110609786, 0.07451061248698075, 0.09842298860734516,
            0.1141645137766335, -0.7955634527928438, -0.14500268078226314,
            0.354022616149075, 0.041061000856168456, 0.10785552005399449,
            -0.5415712728248607, 0.6035480101677293, 0.7021690875827656,
            -0.3941635738323114, 0.32584851988916586, -0.8233278529942015,
            -0.8347330465768846, -0.8511847300819275, -1.3496747989019406,
            -1.3496747989019406, -1.4356920284857626, -2.4990274313385874,
            -0.9509757027657457
          ],
          names: [
            17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33,
            34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50,
            51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67,
            68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84,
            85, 86, 87, 88, 90
          ],
          scores: [
            -1.8394867837743023, -1.8394867837743023, -1.8394867837743023,
            -1.8394867837743023, -1.8144882808205889, -1.6808106825497422,
            -1.6781994732095766, -1.2802650534495061, -0.7291586593824163,
            -0.6839453897108961, -0.4385871297182792, -0.34183312793790577,
            -0.10487902667692617, 0.05656947599434542, 0.04741102318828205,
            0.044443354562768045, 0.10335175041489708, 0.31502944354132256,
            0.24577978273148826, 0.37861130250891883, 0.48489719978592294,
            0.5014569481609579, 0.5456379915872044, 0.5798086235458888,
            0.5266266822690725, 0.5256009495874245, 0.4219032159072092,
            0.6229795133134586, 0.7092018488444858, 0.768144142848028,
            0.8686145159982459, 0.8675197027403536, 0.8188290659631852,
            0.8232723651553442, 0.768274561438148, 0.971175852400557,
            0.9745609933534005, 0.8509678035761017, 0.6756121137260134,
            0.6834991951061402, 0.6800419132304523, 0.6616866505832733,
            0.882034873125356, 0.7189085483667448, 0.6208469686287156,
            0.42550347835400626, 0.33703722438329, 0.3333268662954265,
            0.09086399890046792, 0.05707722213493149, 0.06638742699060002,
            0.23568362987903893, 0.17159856489475447, 0.19140510695192425,
            0.2375786464301802, -0.48492023051688016, 0.1116891047961428,
            0.6057097663237716, 0.18593757152563212, 0.23488404048873568,
            -0.07196558064991193, 0.8720398673077415, 0.9362571035377273,
            -0.08350776771542157, 0.5619761496098077, -0.26884179226251026,
            -0.2729040971308726, -0.2749799054367299, -0.931002188969387,
            -0.931002188969387, -0.9847462572029867, -1.905966243922617,
            -0.44107442479344383
          ],
          type: "univariate",
          upper_bounds: [
            -1.715900670053024, -1.715900670053024, -1.715900670053024,
            -1.715900670053024, -1.713566645049801, -1.5810072310591277,
            -1.5799249607567587, -1.1504857795272454, -0.6816300175022736,
            -0.6326080111439858, -0.38757076542488006, -0.2895291232613181,
            -0.057837478660180956, 0.09712468080515585, 0.08353889291901653,
            0.06851251477150658, 0.14072458064487775, 0.3719442223606204,
            0.2896573602171856, 0.4274811378983675, 0.5251652593492574,
            0.5227033340088958, 0.5772224196843911, 0.6061984918962631,
            0.5477228098517626, 0.5459380765505588, 0.46488327705859434,
            0.6634803296134295, 0.7521585440173688, 0.805461329906564,
            0.9033165768976094, 0.9009574597229597, 0.8471825105592474,
            0.8512077848302918, 0.8094009121496847, 1.0171048040063069,
            1.0148420961149143, 0.9149346401097865, 0.7084013506172587,
            0.7140673217407563, 0.7173154374838812, 0.706018473419493,
            0.953344626933778, 0.8011099246715225, 0.7026503949892076,
            0.484581962045692, 0.41810490146974016, 0.4194698115544821,
            0.16310795196078853, 0.1480527161161353, 0.14838458527311302,
            0.33560691865198, 0.2686865173025282, 0.28438722529650334,
            0.3609927790837269, -0.17427700824091652, 0.36838089037454874,
            0.8573969164984683, 0.33081414219509575, 0.3619125609234769,
            0.3976401115250368, 1.1405317244477537, 1.1703451194926888,
            0.22714803840146824, 0.7981037793304495, 0.285644268469181,
            0.2889248523151394, 0.3012249192084677, -0.5123295790368334,
            -0.5123295790368334, -0.5338004859202108, -1.3129050565066467,
            0.06882685317885806
          ]
        },
        {
          lower_bounds: [
            -0.3356095633469319, 0.46580153913930195, -0.23031554388733386,
            -1.9964517020106922, 0.05530326443770986, 0.26362900304202336,
            -0.42850995221585003, -0.27021185213627963, -3.6051553453424456
          ],
          names: [
            " ?",
            " Federal-gov",
            " Local-gov",
            " Never-worked",
            " Private",
            " Self-emp-inc",
            " Self-emp-not-inc",
            " State-gov",
            " Without-pay",
            "nan"
          ],
          scores: [
            -0.261537931606147, 0.505256245742123, -0.20287525330861667,
            -1.468914588048272, 0.06335549209667879, 0.30902912091644374,
            -0.39400618994916636, -0.23152207680884612, -3.156509202928747
          ],
          type: "univariate",
          upper_bounds: [
            -0.18746629986536212, 0.5447109523449442, -0.17543496272989947,
            -0.9413774740858516, 0.07140771975564772, 0.35442923879086413,
            -0.3595024276824827, -0.19283230148141262, -2.707863060515048
          ]
        }
      ]
    }
  },
  predictedY: [0, 0],
  testData: [
    [20, "Private"],
    [35, "Private"]
  ]
};
